package com.example.mq.mqserver.datacenter;

import com.example.mq.MqApplication;
import com.example.mq.mqserver.core.Binding;
import com.example.mq.mqserver.core.Exchange;
import com.example.mq.mqserver.core.ExchangeType;
import com.example.mq.mqserver.core.MSGQueue;
import com.example.mq.mqserver.mapper.MetaMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;

import javax.annotation.Resource;
import java.io.File;
import java.io.FileFilter;
import java.util.List;

// 通过这个类来整合数据库操作
@Slf4j
public class DataBaseManager {
    // 从 Spring 中拿到现成的对象
    // 因为不想对本类添加 @Component 注解, 因此需要用其他方式获取 Bean (比如 Spring 上下文中)
    private MetaMapper metaMapper;

    // 初始化数据库 = 建库建表 + 插入一些默认数据
    public void init() {
        // 手动的获取到 MetaMapper
        metaMapper = MqApplication.context.getBean(MetaMapper.class);

        // 检查数据库是否已存在
        if(checkDBExists()) {
            log.info("[DataBaseManager] 数据库已存在, 无需初始化!");
            return ;
        }

        // 创建数据库所在目录
        File dataDir = new File("./data");
        dataDir.mkdirs();
        // 创建数据库
        createTable();
        // 插入默认数据
        createDefaultData();
        log.info("[DataBaseManager] 数据库初始化完成!");
    }

    // 删库操作
    public void deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if(ret) {
            log.info("[DataBaseManager] 删除数据库文件成功!");
        }else {
            log.info("[DataBaseManager] 删除数据库文件失败!");
        }

        File dataDir = new File("./data");
        ret = dataDir.delete();
        if(ret) {
            log.info("[DataBaseManager] 删除数据库目录成功!");
        }else{
            log.info("[DataBaseManager] 删除数据库目录失败!");
        }
    }

    // 检查数据库是否已存在
    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        if(file.exists()) {
            return true;
        }
        return false;
    }

    // 该方法用来建库
    // 建库操作不需要手动执行(不需要手动创建 meta.db 文件)
    // 首次执行到这里的数据库操作时, 就会自动的创建 meta.db 文件来 (Mybatis 帮我们完成)
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        log.info("[DataBaseManager] 创建表完成!");
    }

    // 往数据库表中,添加默认数据
    // 此处是添加一个默认的交换机 (匿名交换机, 类型是 DIRECT)
    private void createDefaultData() {
        // 构造一个默认的交换机
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        log.info("[DataBaseManager] 插入默认数据完成!");
    }

    // 封装一下其他的数据库操作
    // --------------------------------------
    public void insertExchange(Exchange exchange) {
        metaMapper.insertExchange(exchange);
    }

    public List<Exchange> selectAllExchanges() {
        return metaMapper.selectAllExchanges();
    }

    public void deleteExchange(String exchangeName) {
        metaMapper.deleteExchange(exchangeName);
    }

    // --------------------------------------
    public void insertQueue(MSGQueue queue) {
        metaMapper.insertQueue(queue);
    }

    public List<MSGQueue> selectAllQueues() {
        return metaMapper.selectAllQueues();
    }

    public void deleteQueue(String queueName) {
        metaMapper.deleteQueue(queueName);
    }

    // --------------------------------------
    public void insertBinding(Binding binding) {
        metaMapper.insertBinding(binding);
    }

    public List<Binding> selectAllBindings() {
        return metaMapper.selectAllBindings();
    }

    public void deleteBinding(Binding binding) {
        metaMapper.deleteBinding(binding);
    }
}
